package com.spzx.product.helper;

import com.spzx.product.api.domain.vo.CategoryVo;

import java.util.ArrayList;
import java.util.List;

public class CategoryHelper {

    /**
     * 使用递归方法建分类树
     * @param categoryVoList
     * @return
     */
    public static List<CategoryVo> buildTree(List<CategoryVo> categoryVoList) {
        List<CategoryVo> trees = new ArrayList<>();
        for (CategoryVo categoryVo : categoryVoList) {
            if (categoryVo.getParentId().longValue() == 0) {
                //在categoryVoList中找到categoryVo的子节点，并组装到categoryVo中
                categoryVo = findAndAddChildren(categoryVo, categoryVoList);
                trees.add(categoryVo);
            }
        }
        return trees;
    }

    /**
     * 在categoryVoList中找到categoryVo的子节点，并组装到categoryVo中
     * @param categoryVo
     * @param categoryVoList
     * @return 返回组装好的categoryVo节点
     */
    public static CategoryVo findAndAddChildren(CategoryVo categoryVo, List<CategoryVo> categoryVoList) {
        for (CategoryVo node : categoryVoList) {
            if(node.getParentId().longValue() == categoryVo.getId().longValue()) {
                //在categoryVoList中找到categoryVo的子节点，
                CategoryVo children = findAndAddChildren(node, categoryVoList);
                //并组装到categoryVo中
                categoryVo.getChildren().add(children);
            }
        }
        return categoryVo;
    }
}
